mpirun -n 8 python train.py --config ./src/configs/small_cswin_224.yaml > train_1.log 2>&1 &